import os
# 导入OS包


def count_file_size(dir1):
    dlist = os.listdir(dir1)  # 获取路径下的文件目录
    file_size = 0

    for f in dlist:
        file = os.path.join(dir1, f)  # 遍历目录中的文件（夹）并生成其绝对路径

        if os.path.isfile(file):  # 如果是文件就获得其大小
            file_size += os.path.getsize(file)
        if os.path.isdir(file):
            file_size += count_file_size(file)  # 递归调用

    return file_size


path = input('请输入文件路径：')
print('File size in ', os.getcwd(), '=', count_file_size(path) / 1024 / 1024, 'MB')
